# 主要是给MOTR用的，避免他Eval的时候加载txt文件

import os
import glob

base_dir = "/home/konghaomin/Datasets/MaritimeTrack_20250322_D"

# 遍历所有txt文件
txt_files = glob.glob(os.path.join(base_dir, "**/*.txt"), recursive=True)

# 计数器
deleted_count = 0
kept_count = 0

# 处理每个txt文件
for txt_file in txt_files:
    # 获取文件名
    file_name = os.path.basename(txt_file)

    # 如果文件名不是gt.txt，则删除
    if file_name != "gt.txt" and "seqmap" not in file_name:
        os.remove(txt_file)
        print(f"已删除: {txt_file}")
        deleted_count += 1
    else:
        kept_count += 1

print(f"删除完成! 共删除了 {deleted_count} 个文件，保留了 {kept_count} 个gt.txt文件。")
